Python Program Interview Question